module.exports = function (source) {
  let newSource = source;
  if (newSource.indexOf("from 'umi'") !== -1) {
    newSource = newSource.replace("from 'umi'", "from '@umijs/max'");
  }
  return newSource;
};
